import torch
import torch.nn as nn
import torch.optim as optim
import numpy as np

if __name__ == '__main__':
    use_cuda = True
    device = torch.device("cuda" if torch.cuda.is_available() and use_cuda else "cpu")

    image_size = [3, 40, 90]
    x = torch.zeros([1, image_size[0], image_size[1], image_size[2]]).to(device)
    print(x)



    1


